package com.kfyang.manualgear.generator.impl.jdbc;

import static com.kfyang.manualgear.constant.Constants.*;
import static com.kfyang.manualgear.constant.StatementConstants.*;
import static com.kfyang.manualgear.generator.impl.jdbc.constant.StatementConstants.JDBC_MODIFY_STATEMENT;

import com.kfyang.manualgear.generator.declare.ClassWrapper;
import com.kfyang.manualgear.generator.declare.ModifyMethodGenerator;

import javax.lang.model.element.ExecutableElement;

public class ModifyMethodGeneratorJdbcImpl extends ModifyMethodGenerator {

    public ModifyMethodGeneratorJdbcImpl(ExecutableElement element, ClassWrapper classWrapper, boolean useSimple) {
        super(element, classWrapper, useSimple);
    }

    @Override
    protected void addReturn() {
        methodBuilder.addStatement((hasReturn ? RETURN : EMPTY) + JDBC_MODIFY_STATEMENT);
    }
}
